Python interview preparation